package com.briup.oagg.model.bean.op;

import com.briup.framework.utils.common.StringUtil;
import lombok.Data;

import java.util.Base64;
import java.util.Map;

/**
 *  训练结果类
 */
@Data
public class TaskResult {
    private String task_id;   //模型训练配置编号对应ModelTrainConfig类中trainId属性
    private String task_type; //训练模型类型
    private int status;      //训练状态
    private String preprocessor;//对应ModelTrainConfig类中preprocessWeight属性
    private String ml;//对应ModelTrainConfig类中algorithmWeight属性
    private Map<String,Object> eva; //训练结果
    private String msg;//如果训练失败，返回错误提示信息，如果训练成功，不返回

    /**
     *
     * @return
     */
    public byte[] getPreprocessWeight() {
        if(StringUtil.isNullOrEmpty(preprocessor)) return null;
        return Base64.getDecoder().decode(preprocessor);
    }

    /**
     *
     * @return
     */
    public byte[] getAlgorithmWeight() {
        if(StringUtil.isNullOrEmpty(preprocessor)) return null;
        return Base64.getDecoder().decode(ml);
    }
}
